package com.linran.structure_algorithm.算法.lesson3_查找;

import java.util.Collection;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * 递归方式二分查找
 *
 * 注意：使用二分查找的前提是该数组是有序的.
 */
public class BinaryRecursiveSearch {

    public static void main(String[] args) {
        int[] arr = {1, 8, 10, 89, 1000, 1000, 1234};
        int targetIndex = binarySearch(arr, 0, arr.length, 10);
        if (targetIndex == -1) {
            System.out.println("未找到目标值");
            return;
        }
        System.out.println(arr[targetIndex]);

//        Collection<Integer> integers = binaryMutilSearch(arr, 0, arr.length - 1, 1000);
//        if (Objects.nonNull(integers)) {
//            integers.forEach(index -> System.out.println(index + ":" + arr[index]));
//        }
    }

    /**
     * 二分查找
     *
     * @param arr     数组
     * @param left    左边索引
     * @param right   右边索引
     * @param findVal 目标查询值
     * @return 如果找到就返回下标，如果没有找到就返回-1
     */
    public static int binarySearch(int[] arr, int left, int right, int findVal) {
        if (left > right) {
            return -1;
        }

        int mid = (left + right) / 2;
        if (findVal < arr[mid]) {
            // 向左递归
            return binarySearch(arr, left, mid - 1, findVal);
        } else if (findVal > arr[mid]) {
            // 向右递归
            return binarySearch(arr, mid + 1, right, findVal);
        } else {
            return mid;
        }
    }

    /**
     * 二分查找
     *
     * @param arr     数组
     * @param left    左边索引
     * @param right   右边索引
     * @param findVal 目标查询值
     * @return 如果找到就返回下标，如果没有找到就返回寻找到的多个下标
     */
    public static Collection<Integer> binaryMutilSearch(int[] arr, int left, int right, int findVal) {
        if (left > right) {
            return null;
        }

        int mid = (left + right) / 2;
        if (findVal < arr[mid]) {
            // 向左递归
            return binaryMutilSearch(arr, left, mid - 1, findVal);
        } else if (findVal > arr[mid]) {
            // 向右递归
            return binaryMutilSearch(arr, mid + 1, right, findVal);
        } else {
            // 找到目标值
            // 继续左递归
            Collection<Integer> leftResult = binaryMutilSearch(arr, left, mid - 1, findVal);
            // 继续右递归
            Collection<Integer> rightResult = binaryMutilSearch(arr, mid + 1, right, findVal);
            return Stream.of(Set.of(mid), leftResult, rightResult).filter(Objects::nonNull).flatMap(Collection::stream).collect(Collectors.toSet());
        }
    }
}
